"""
@Time : 2023/12/19 14:53
@Auth : Mr.Zhao
"""
#!/usr/bin/env python
# coding:utf-8

from hashlib import md5

import requests


class Chaojiying_Client(object):

    def __init__(self, username = 'jingluozhihai', password = '97f1b81115be21df1c60fec334cd2ef1', soft_id = 956297):
        self.username = username
        self.password = password
        #下面两行将明文加密为md5，若不想直接传递明文导致密码泄漏，可以直接在外面传入md5加密后的数据
        # password =  password.encode('utf8')
        # self.password = md5(password).hexdigest()
        self.soft_id = soft_id
        self.base_params = {
            'user': self.username,
            'pass2': self.password,
            'softid': self.soft_id,
        }
        self.headers = {
            'Connection': 'Keep-Alive',
            'User-Agent': 'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)',
        }

    def PostPic(self, im, codetype = 1902):
        """
        im: 图片字节
        codetype: 题目类型 参考 http://www.chaojiying.com/price.html
        """
        im = open(im, 'rb').read()	  #我直接写到函数内部,获取文件的字节码（即图片对应的二进制码）

        params = {
            'codetype': codetype,
        }
        params.update(self.base_params) #读取图片文二进制
        files = {'userfile': ('ccc.jpg', im)} #这个'ccc.jpg'不知道有啥用，好像啥也不影响
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, files=files, headers=self.headers)
        #return r.json()
        return (r.json())['pic_str'] #自己加的--只要验证码

    def PostPic_base64(self, base64_str, codetype = 1902):
        """
        im: 图片字节
        codetype: 题目类型 参考 http://www.chaojiying.com/price.html
        """
        params = {
            'codetype': codetype,
            'file_base64':base64_str
        }
        params.update(self.base_params)
        r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, headers=self.headers)
        #return r.json()
        return (r.json())['pic_str'] #自己加的--只要验证码

    def ReportError(self, im_id):
        """
        im_id:报错题目的图片ID
        """
        params = {
            'id': im_id,
        }
        params.update(self.base_params)
        r = requests.post('http://upload.chaojiying.net/Upload/ReportError.php', data=params, headers=self.headers)
        return r.json()


if __name__ == '__main__':
    #加密
    # password = "xxxxxxxxxxx"
    # password =  password.encode('utf8')
    # password = md5(password).hexdigest()
    #print(password) #97f1b81115be21df1c60fec334cd2ef1
    chaojiying = Chaojiying_Client('jingluozhihai', '97f1b81115be21df1c60fec334cd2ef1', '956297')	#用户中心>>软件ID 生成一个替换 96001
    #我要把im放到函数内部，这样就懒得在外面调用了，直接穿函数就行了
    #im = open('a.png', 'rb').read()	   #本地图片文件路径 来替换 a.png 有时WIN系统须要//
    ret = chaojiying.PostPic('a.png', 1902)  #1902 验证码类型  官方网站>>价格体系 3.4+版 print 后要加()
    print(ret)
    #print(ret['pic_str'])
    #print(chaojiying.PostPic('./a.png', 1902))  #直接传图片地址
    #print(chaojiying.PostPic(base64_str, 1902)) #此处为传入 base64代码
